package com.example.mq.mqserver.core;

import com.example.mq.common.ConsumerEnv;
import com.fasterxml.jackson.core.JsonProcessingException;
import com.fasterxml.jackson.core.type.TypeReference;
import com.fasterxml.jackson.databind.ObjectMapper;
import lombok.Data;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.concurrent.atomic.AtomicInteger;

/**
 * 存储消息的队列
 */
@Data
public class MSGQueue {

    /**
     * 队列的身份标识
     */
    private String name;

    /**
     * 队列是否持久化
     * true 持久化
     * false 不持久化
     */
    private boolean durable = false;

    /**
     * 独占
     * true 这个队列只能被一个消费者使用
     * false 大家都能使用
     */
    private boolean exclusive = false;

    /**
     * 自动删除
     * true 没人使用自动删除
     * false 不会自动删除
     */
    private boolean autoDelete = false;

    /**
     * 扩展参数
     * 为了把 arguments 存到数据库中，需要把 Map 转成 json 格式的字符串
     */
    private Map<String, Object> arguments = new HashMap<>();

    /**
     * 当前队列有哪些消费者订阅了
     */
    private List<ConsumerEnv> consumerEnvList = new ArrayList<>();

    /**
     * 记录当前取到了第几个消费者，方便实现查询
     * AtomicInteger 原子类型变量，线程安全
     */
    private AtomicInteger consumerSeq = new AtomicInteger(0);



    // 添加新的订阅者
    public void addConsumerEnv(ConsumerEnv consumerEnv){
        synchronized (this){
            consumerEnvList.add(consumerEnv);
        }
    }

    // 挑选一个订阅者，用来处理当前的消息，（轮询）
    public ConsumerEnv chooseConsumer(){
        if(0 == consumerEnvList.size()){
            // 该队列没有人订阅
            return null;
        }
        // 计算当前要取的元素的下标
        int index = consumerSeq.get() % consumerEnvList.size();
        // 自增 1
        consumerSeq.getAndIncrement();
        return consumerEnvList.get(index);
    }


    public String getArguments(){
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            return objectMapper.writeValueAsString(arguments);
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
        return "{}";
    }

    public void setArguments(String argumentsJson){
        ObjectMapper objectMapper = new ObjectMapper();
        try {
            this.arguments = objectMapper.readValue(argumentsJson, new TypeReference<HashMap<String, Object>>() {
            });
        } catch (JsonProcessingException e) {
            e.printStackTrace();
        }
    }

    public Object getArguments(String key){
        return arguments.get(key);
    }

    public void setArguments(String key, Object value){
        arguments.put(key, value);
    }

    public void setArguments(Map<String, Object> arguments){
        this.arguments = arguments;
    }
}
